
from flask import Flask, request, jsonify
from openai import OpenAI

app = Flask(__name__, static_folder='static')

client = OpenAI(
    base_url="https://ai.gitee.com/v1",
    api_key="UVLCTEKMREDSE19EZ7DMVCLF1ZVUZA2OZRPVIO1O",
    default_headers={"X-Failover-Enabled":"true"},
)

@app.route('/ask', methods=['POST'])
def ask():
    data = request.json
    question = data.get('question', '') if data is not None else ''

    response = client.chat.completions.create(
        model="QwQ-32B",
        stream=False,
        max_tokens=5000,
        temperature=1,
        top_p=0.2,
        extra_body={"top_k": 50},
        frequency_penalty=0,
        messages=[
            {
                "role": "system",
                "content": "你是一名熟知历代名人的老师，对于他们的身世，故事，以及评价非常清晰，用来帮助用户以名人为榜样，或者借鉴来提升自己。先获取用户的需求，会告诉你他想了解的是哪位名人，你要先说出他的身世，然后说故事，和精彩的时候，最后说一下用户应该从名人身上学到什么品质。逻辑性要非常的强，不要用LaJex\n格式返回，使用is的Matby\nax库来表示。最终只给出H\nTML与CSS还有JS代码完成\n的故事介绍和分析。"
            },
            {
                "role": "user",
                "content": question
            }
        ],
    )

    answer = response.choices[0].message.content
    return jsonify({'answer': answer})

@app.route('/')
def index():
    try:
        return app.send_static_file('index.html')
    except Exception as e:
        return f"文件未找到: {e}. 请确保 'index.html' 位于 'static' 文件夹中。", 404

if __name__ == '__main__':
    app.run(debug=True, port=5000)